import torch as t
import time
a = t.rand(10000,10000)
b = t.rand(10000,10000)
start1 = time.time()
a.matmul(b)
end1 = time.time()-start1
print("cup",end1)

a = a.cuda()
b = b.cuda()
start2 = time.time()
a.matmul(b)
end2 = time.time()-start2
print("gpu",end2)